tensor([[[[[0.2500, 0.2500],
           [0.2500, 0.2500]],

          [[0.2500, 0.2500],
           [0.2500, 0.2500]]],


         [[[0.2500, 0.2500],
           [0.2500, 0.2500]],

          [[0.2500, 0.2500],
           [0.2500, 0.2500]]]],



        [[[[0.2800, 0.2800],
           [0.2800, 0.2800]],

          [[0.2800, 0.2800],
           [0.2800, 0.2800]]],


         [[[0.2800, 0.2800],
           [0.2800, 0.2800]],

          [[0.2800, 0.2800],
           [0.2800, 0.2800]]]],



        [[[[0.2400, 0.2400],
           [0.2400, 0.2400]],

          [[0.2400, 0.2400],
           [0.2400, 0.2400]]],


         [[[0.2400, 0.2400],
           [0.2400, 0.2400]],

          [[0.2400, 0.2400],
           [0.2400, 0.2400]]]],



        [[[[0.2200, 0.2200],
           [0.2200, 0.2200]],

          [[0.2200, 0.2200],
           [0.2200, 0.2200]]],


         [[[0.2200, 0.2200],
           [0.2200, 0.2200]],

          [[0.2200, 0.2200],
           [0.2200, 0.2200]]]]], grad_fn=<RoundBackward1>)